package com.niit.ai.dl.dltrain.contants;

public enum EModelType {
    MNIST(1), FASHION(2), FLOWER(3), TITANIC(4), NONLINER_SYSTEM(5);

    private int id;

    EModelType(int id) {
        this.id = id;
    }

    public int getId() {
        return id;
    }

    public static EModelType getEnumById(int id) {
        for (EModelType modelType : EModelType.values()) {
            if (modelType.id == id) {
                return modelType;
            }
        }
        return null;
    }
}
